package com.uptang.paper.main;

import lombok.Data;
import lombok.extern.slf4j.Slf4j;

import javax.imageio.stream.FileImageOutputStream;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

@Slf4j
@Data
public class ImageParse {

    public String parse(ByteArrayInputStream byteArrayInputStream, String fullName) {
        String s = null;
        try {
            s = saveBit(byteArrayInputStream, fullName);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return s;
    }

    public String saveBit(InputStream inStream, String fullName) throws IOException {
        String fileName = System.currentTimeMillis() + fullName;
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        //创建一个Buffer字符串
        byte[] buffer = new byte[1024];
        //每次读取的字符串长度，如果为-1，代表全部读取完毕
        int len = 0;
        //使用一个输入流从buffer里把数据读取出来
        while ((len = inStream.read(buffer)) != -1) {
        //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
            outStream.write(buffer, 0, len);
        }
        //关闭输入流
        inStream.close();
        //把outStream里的数据写入内存

        //得到图片的二进制数据，以二进制封装得到数据，具有通用性
        byte[] data = outStream.toByteArray();
        //new一个文件对象用来保存图片，默认保存当前工程根目录
        File imageFile = new File("F:/image/" + fileName);
        //创建输出流
        FileOutputStream fileOutStream = new FileOutputStream(imageFile);
        //写入数据
        fileOutStream.write(data);
        return fileName;
    }


    public String parse(byte[] data, String fullName) {
        String fileName = System.currentTimeMillis() + ".png";
        String path = "F:\\image\\" + fileName;
        if (data.length < 3) {
            return null;
        }
        try {
            FileImageOutputStream imageOutput = new FileImageOutputStream(new File(path));
            imageOutput.write(data, 0, data.length);
            imageOutput.close();
        } catch (Exception ex) {
            log.error("Exception: ", ex);
            ex.printStackTrace();
        }
        return fileName;
    }
}
